package cn.com.nes.site.service.fault.impl;

import cn.com.nes.mybatis.agent.ope.entity.IsFaultTypeInfo;
import cn.com.nes.mybatis.agent.ope.mapper.IsFaultTypeInfoMapper;
import cn.com.nes.site.entity.dto.fault.GetFaultListByProtocolIdDto;
import cn.com.nes.site.service.fault.FaultTypeService;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.List;

@Service
public class FaultTypeServiceImpl implements FaultTypeService {

    @Resource
    private IsFaultTypeInfoMapper faultTypeInfoMapper;

    @Override
    public List<GetFaultListByProtocolIdDto> getFaultType(String manufactureId, Byte isAgent) {
        List<IsFaultTypeInfo> results = faultTypeInfoMapper.selectList(null);
        List<GetFaultListByProtocolIdDto> dtos  = faultTypeInfoMapper.getAllFaultTypes(manufactureId, isAgent);
        for (IsFaultTypeInfo res : results) {
            if (res.getManufactureId().equals(manufactureId)) {
                GetFaultListByProtocolIdDto dto = new GetFaultListByProtocolIdDto();
                dto.setFaultTypeId(res.getFaultTypeId());
                dto.setFaultTypeName(res.getFaultTypeName());
                dtos.add(dto);
            }
        }
        return dtos;
    }

}
